package com.dev.toolkit.config.ds;

/**
 * @author ZhuYouBin
 * @version 1.0.0
 * @date 2024/5/21 11:40
 * @description 数据源线程缓存类
 */
public class DsContextHolder {
    private static final ThreadLocal<String> HOLDER = new ThreadLocal<>();

    /**
     * 切换数据源类型
     *
     * @param dataSourceType 数据源的 bean 名称
     */
    public static void switchDataSourceType(String dataSourceType) {
        HOLDER.set(dataSourceType);
    }

    /**
     * 获取当前数据源的key名称
     */
    public static String getDataSourceType() {
        return HOLDER.get() == null ? DsMyBatisConfig.DEFAULT_DATA_SOURCE_NAME : HOLDER.get();
    }

    /**
     * 清空数据源
     */
    public static void clearDataSourceType() {
        HOLDER.remove();
    }
}